Machine learning Pipeline in python